import torch #必须先导入torch
import ops_test_op

#tensor是函数传值,大写Tensor是类传shape
input1 = torch.tensor([[0,1,2],[3,4,5]]).to(torch.float32)
input2 = torch.tensor([[100,100,100],[100,100,100]]).to(torch.float32)
output = torch.tensor([[0.0,0.0,0.0],[0.0,0.0,0.0]]).to(torch.float32)

ops_test_op.test(input1,input2,output)
print(output)
pass